// /* Copyright 2018 The TensorFlow Authors. All Rights Reserved.

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at

//     http://www.apache.org/licenses/LICENSE-2.0

// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// ==============================================================================*/

// #include "tao_bridge/tf/xla_cluster_util.h"

// // #include "tensorflow/cc/framework/ops.h"
// // #include "tensorflow/cc/ops/control_flow_ops_internal.h"
// // #include "tensorflow/cc/ops/standard_ops.h"
// // #include "tensorflow/core/framework/function_testlib.h"
// #include "tensorflow/core/framework/graph_to_functiondef.h"
// #include "tensorflow/core/graph/algorithm.h"
// #include "tensorflow/core/graph/graph_constructor.h"
// #include "tensorflow/core/graph/graph_def_builder.h"
// // #include "tensorflow/core/graph/testlib.h"
// // #include "tensorflow/core/lib/core/status_test_util.h"
// // #include "tensorflow/core/platform/test.h"

// #include "tao_bridge/test_helpers.h"

// namespace tensorflow {
// namespace tao {
// namespace {

// TEST(CreateCycleDetectionGraph, ConnectivityThroughEnterExitRegion) {
//   Scope root = Scope::NewRootScope().ExitOnError();

//   Output a = ops::Const(root.WithOpName("a"), Input::Initializer(0.0));
//   Output enter =
//       ops::internal::Enter(root.WithOpName("enter"), a, "only_frame");
//   Output exit = ops::internal::Exit(root.WithOpName("exit"), enter);
//   Output b = ops::Add(root.WithOpName("b"), a, exit);

//   FixupSourceAndSinkEdges(root.graph());

//   GraphCycles cycles;
//   TF_ASSERT_OK(CreateCycleDetectionGraph(root.graph(), &cycles));
//   EXPECT_FALSE(cycles.ContractEdge(a.node()->id(), b.node()->id()));
// }

// TEST(CreateCycleDetectionGraph, ConnectivityThroughMultipleEnterExitRegions)
// {
//   Scope root = Scope::NewRootScope().ExitOnError();

//   Output a = ops::Const(root.WithOpName("a"), Input::Initializer(0.0));
//   Output enter_0 =
//       ops::internal::Enter(root.WithOpName("enter_0"), a, "frame_0");
//   Output exit_0 = ops::internal::Exit(root.WithOpName("exit_0"), enter_0);
//   Output enter_1 =
//       ops::internal::Enter(root.WithOpName("enter_1"), a, "frame_1");
//   Output exit_1 = ops::internal::Exit(root.WithOpName("exit_1"), enter_1);
//   Output b = ops::Add(root.WithOpName("b"), a, exit_1);

//   FixupSourceAndSinkEdges(root.graph());

//   GraphCycles cycles;
//   TF_ASSERT_OK(CreateCycleDetectionGraph(root.graph(), &cycles));
//   EXPECT_FALSE(cycles.ContractEdge(a.node()->id(), b.node()->id()));
// }
// }  // namespace
// }  // namespace tao
// }  // namespace tensorflow
